colab에서는 분명 잘 되었던 건데 윈도우에서 자체 세팅한 jupyter로 옮기고 실행하다가 해당 에러를 만나셨나요?
윈도우에서는 torch.utils.data.DataLoader()의 num_workers 값이 0이 아니면 해당 에러가 나타날 수 있습니다.
아래처럼 값을 0으로 변경해주면 정상적으로 동작하게 됩니다.
torch.utils.data.DataLoader(train_data, batch_size=batch_size, num_workers=0)